from inspect import signature
from functools import wraps


def typeassert(*ty_args, **ty_kwargs):
    def decorate(func):
        # 如果在优化模式下，禁用类型检查
        if not __debug__:  # python -O xx.py就是not __debug__
            return func
        # 将函数参数名称映射到提供的类型
        sig = signature(func)

        # bind_partial和functools.partical差不多部分绑定，arguments返回的是OrderedDict([('value', 'zhangsan')])
        bound_types = sig.bind_partial(*ty_args, **ty_kwargs).arguments  # 装饰器指定的类型

        @wraps(func)
        def wrapper(*args, **kwargs):
            bound_values = sig.bind(*args, **kwargs)  # 函数实际的类型
            # 在提供的参数之间执行类型断言¬
            for name, value in bound_values.arguments.items():
                if name in bound_types:
                    if not isinstance(value, bound_types[name]):
                        raise TypeError(
                            'Argument {} must be {}'.format(name, bound_types[name])
                        )
            return func(*args, **kwargs)

        return wrapper

    return decorate


@typeassert(z=int)
def spam(x, y, z=42):
    print(x, y, z)


if __name__ == '__main__':
    spam(1, 2, 3)
    #spam(1, 'hello', 'world')
